package database

import (
	"database/sql"

	"github.com/alecthomas/log4go"
	_ "github.com/go-sql-driver/mysql"
)

var db *sql.DB

// InitDB 初始化数据库
func InitDB(dsn string) error {
	log4go.Info("connect to mysql...")

	var err error
	db, err = sql.Open("mysql", dsn)
	if err != nil {
		log4go.Error(err)
		return err
	}

	db.SetMaxOpenConns(100)
	db.SetMaxIdleConns(50)

	log4go.Info("succeed to connect to mysql")

	return nil
}

// GetDb 取数据库连接
func GetDb() *sql.DB {
	return db
}
